python add a third variable with color